"""
内容:使用pandas将文件中的数据取出
作者:刘芳
"""
# 引入模块
import pandas
import os

# 创建一个pandas数据操作类
class PandasData(object):
    # 定义属性
    def __init__(self,file_name):
        data_path = os.path.dirname(os.path.dirname(__file__)) + f"/data/{file_name}"  # 拼接路径
        self.table = pandas.read_excel(data_path, keep_default_na=False)

    # 定义方法
    """将数据转换成列表套字典"""
    def pandas_list_dict(self):
        list_data = []
        for i in self.table.index.values:
            list_data.append(self.table.loc[i].to_dict())
        return list_data

    """将数据转换成列表套列表"""
    def pandas_list_list(self):
        list_data = []
        for i in self.table.index.values:
            list_data.append(self.table.loc[i].to_list())
        return list_data

if __name__ == '__main__':
    pd1 = PandasData("search_input_data.xls")
    print(pd1.pandas_list_dict())
    print(pd1.pandas_list_list())
